package com.yunxi.module.system.service.permission.impl;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ObjectUtil;
import com.google.common.annotations.VisibleForTesting;
import com.mzt.logapi.context.LogRecordContext;
import com.mzt.logapi.starter.annotation.LogRecord;
import com.yunxi.framework.common.enums.CommonStatusEnum;
import com.yunxi.framework.common.exception.util.ServiceExceptionUtil;
import com.yunxi.framework.common.pojo.PageResult;
import com.yunxi.framework.common.util.collection.CollectionUtils;
import com.yunxi.framework.common.util.object.BeanUtils;
import com.yunxi.framework.common.util.spring.SpringUtils;
import com.yunxi.module.system.enums.constant.ErrorCodeConstants;
import com.yunxi.module.system.enums.constant.LogRecordConstants;
import com.yunxi.module.system.enums.permission.DataScopeEnum;
import com.yunxi.module.system.enums.permission.RoleCodeEnum;
import com.yunxi.module.system.enums.permission.RoleTypeEnum;
import com.yunxi.module.system.mapper.permission.RoleMapper;
import com.yunxi.module.system.model.permission.RoleDO;
import com.yunxi.module.system.redis.RedisKeyConstants;
import com.yunxi.module.system.service.permission.PermissionService;
import com.yunxi.module.system.service.permission.RoleService;
import com.yunxi.module.system.vo.admin.permission.role.RolePageReqVO;
import com.yunxi.module.system.vo.admin.permission.role.RoleSaveReqVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Set;

/**
 * <p>
 * 角色 Service 实现类
 * </p>
 *
 * @author lidy
 * @since 2024-05-07
 */
@Service
@Slf4j
public class RoleServiceImpl implements RoleService {

    @Resource
    private RoleMapper roleMapper;

    @Resource
    private PermissionService permissionService;

    /**
     * 判断角色编号数组中，是否有管理员
     *
     * @param ids 角色编号数组
     * @return 是否有管理员
     */
    @Override
    public boolean hasAnySuperAdmin(Collection<Long> ids) {
        if (CollectionUtil.isEmpty(ids)) {
            return false;
        }
        RoleServiceImpl self = getSelf();

        return ids.stream().anyMatch(id -> {
            RoleDO roleDO = self.getRoleFromCache(id);
            return roleDO != null && RoleCodeEnum.isSuperAdmin(roleDO.getCode());
        });
    }

    /**
     * 从缓存中获得角色
     *
     * @param id 角色 ID
     * @return 角色
     */
    @Override
    @Cacheable(value = RedisKeyConstants.ROLE, key = "#id", unless = "#result == null ")
    public RoleDO getRoleFromCache(Long id) {
        return roleMapper.selectById(id);
    }

    /**
     * 获得角色列表
     *
     * @param roleIds 角色编号数组
     * @return 角色列表
     */
    @Override
    public List<RoleDO> getRoleList(Collection<Long> roleIds) {
        if (CollectionUtil.isEmpty(roleIds)) {
            return Collections.emptyList();
        }
        return roleMapper.selectBatchIds(roleIds);
    }

    /**
     * 从缓存中获得角色数组
     *
     * @param ids 角色编号数组
     * @return 角色数组
     */
    @Override
    public List<RoleDO> getRoleListFromCache(Collection<Long> ids) {
        if (CollectionUtil.isEmpty(ids)) {
            return Collections.emptyList();
        }

        // 采用 for 循环从缓存中获取，主要考虑 Spring CacheManager 无法批量操作的问题
        RoleServiceImpl self = getSelf();

        return CollectionUtils.convertList(ids, self::getRoleFromCache);
    }

    /**
     * 根据状态查询角色集合
     *
     * @param statuses 状态集合
     * @return 角色集合
     */
    @Override
    public List<RoleDO> getRoleListByStatus(Collection<Integer> statuses) {
        return roleMapper.selectByStatus(statuses);
    }

    /**
     * 获得角色分页列表
     *
     * @param reqVO 分页参数
     * @return 角色分页列表
     */
    @Override
    public PageResult<RoleDO> getRolePage(RolePageReqVO reqVO) {
        return roleMapper.selectPage(reqVO);
    }

    /**
     * 根据角色编号获得角色信息
     *
     * @param roleId 角色编号
     * @return 角色信息
     */
    @Override
    public RoleDO getRole(Long roleId) {
        return roleMapper.selectById(roleId);
    }

    /**
     * 删除角色
     *
     * @param roleId 角色编号
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    @CacheEvict(value = RedisKeyConstants.ROLE, key = "#roleId")
    @LogRecord(type = LogRecordConstants.SYSTEM_ROLE_TYPE, subType = LogRecordConstants.SYSTEM_ROLE_DELETE_SUB_TYPE,
            bizNo = "{{#roleId}}", success = LogRecordConstants.SYSTEM_ROLE_DELETE_SUCCESS)
    public void deleteRole(Long roleId) {
        // 1. 校验是否可以更新
        RoleDO roleDO = validateRoleForUpdate(roleId);

        // 2.1 标记删除
        roleMapper.deleteById(roleId);
        // 2.2 删除相关数据
        permissionService.processRoleDeleted(roleId);

        // 3. 记录操作日志上下文
        LogRecordContext.putVariable("role", roleDO);
    }

    /**
     * 创建角色信息
     *
     * @param createReqVO 角色信息
     * @param type        角色类型
     * @return 角色编号
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    @LogRecord(type = LogRecordConstants.SYSTEM_ROLE_TYPE, subType = LogRecordConstants.SYSTEM_ROLE_CREATE_SUB_TYPE,
            bizNo = "{{#roleDO.id}}", success = LogRecordConstants.SYSTEM_ROLE_CREATE_SUCCESS)
    public Long createRole(RoleSaveReqVO createReqVO, Integer type) {
        // 1. 校验角色是否重复
        validateRoleDuplicate(createReqVO.getName(), createReqVO.getCode(), null);

        // 2. 插入到数据库
        RoleDO roleDO = BeanUtils.toBean(createReqVO, RoleDO.class)
                .setType(ObjectUtil.defaultIfNull(type, RoleTypeEnum.CUSTOM.getType()))
                .setStatus(CommonStatusEnum.ENABLE.getStatus()) // 默认可用
                .setDataScope(DataScopeEnum.ALL.getScope()); // 默认可查看所有数据。原因；避免某些项目不需要限制权限
        roleMapper.insert(roleDO);

        // 3. 记录操作日志上下文
        LogRecordContext.putVariable("role", roleDO);

        return roleDO.getId();
    }

    /**
     * 修改角色
     *
     * @param updateReqVO 角色信息
     */
    @Override
    @CacheEvict(value = RedisKeyConstants.ROLE, key = "#updateReqVO.id")
    @LogRecord(type = LogRecordConstants.SYSTEM_ROLE_TYPE, subType = LogRecordConstants.SYSTEM_ROLE_CREATE_SUB_TYPE,
            bizNo = "{{#updateReqVO.id}}", success = LogRecordConstants.SYSTEM_ROLE_UPDATE_SUCCESS)
    public void updateRole(RoleSaveReqVO updateReqVO) {
        // 1.1 校验是否可以更新
        RoleDO roleDO = validateRoleForUpdate(updateReqVO.getId());

        // 1.2 校验角色的唯一字段是否重复
        validateRoleDuplicate(updateReqVO.getName(), updateReqVO.getCode(), updateReqVO.getId());

        // 2. 更新数据
        RoleDO updateObj = BeanUtils.toBean(updateReqVO, RoleDO.class);
        roleMapper.updateById(updateObj);

        // 3. 记录操作日志上下文
        LogRecordContext.putVariable("role", roleDO);
    }

    /**
     * 修改角色状态
     *
     * @param roleId 角色编号
     * @param status 状态
     */
    @Override
    public void updateRoleStatus(Long roleId, Integer status) {
        // 校验角色是否允许更新
        validateRoleForUpdate(roleId);

        // 更新角色
        RoleDO roleDO = new RoleDO().setId(roleId).setStatus(status);
        roleMapper.updateById(roleDO);
    }

    /**
     * 为角色分配数据权限
     *
     * @param roleId           角色编号
     * @param dataScope        数据权限
     * @param dataScopeDeptIds 数据权限部门编号
     */
    @Override
    @CacheEvict(value = RedisKeyConstants.ROLE, key = "#roleId")
    public void updateRoleDataScope(Long roleId, Integer dataScope, Set<Long> dataScopeDeptIds) {
        // 校验是否可以更新
        validateRoleForUpdate(roleId);

        // 更新数据范围
        RoleDO roleDO = new RoleDO();
        roleDO.setId(roleId);
        roleDO.setDataScope(dataScope);
        roleDO.setDataScopeDeptIds(dataScopeDeptIds);
        roleMapper.updateById(roleDO);
    }

    /**
     * 获得所有角色
     *
     * @return 所有角色列表
     */
    @Override
    public List<RoleDO> getRoleList() {
        return roleMapper.selectList();
    }

    /**
     * 校验角色是否重复：
     * 1. 角色名称是否重复
     * 2. 角色编码是否重复
     *
     * @param name   角色名称
     * @param code   角色编码
     * @param roleId 角色编号
     */
    @VisibleForTesting
    private void validateRoleDuplicate(String name, String code, Long roleId) {
        // 1. 超级管理员，不允许创建
        if (RoleCodeEnum.isSuperAdmin(code)) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.ROLE_ADMIN_CODE_ERROR, code);
        }

        // 2. name 名称被其他角色使用
        RoleDO roleDO = roleMapper.selectByName(name);
        if (roleDO != null && !roleDO.getId().equals(roleId)) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.ROLE_NAME_DUPLICATE, name);
        }

        // 3. 是否存在相同编码的角色
        if (!StringUtils.hasText(code)) {
            return;
        }

        // 4. code 编码被其他角色使用
        roleDO = roleMapper.selectByCode(code);
        if (roleDO != null && !roleDO.getId().equals(roleId)) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.ROLE_CODE_DUPLICATE, code);
        }
    }

    /**
     * 校验角色是否可以更新
     *
     * @param roleId 角色编号
     * @return 角色信息
     */
    @VisibleForTesting
    private RoleDO validateRoleForUpdate(Long roleId) {
        // 根据编号查询角色
        RoleDO roleDO = roleMapper.selectById(roleId);
        if (roleDO == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.ROLE_NOT_EXISTS);
        }

        // 内置角色，不允许删除
        if (roleDO.getType().equals(RoleTypeEnum.SYSTEM.getType())) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.ROLE_CAN_NOT_UPDATE_SYSTEM_TYPE_ROLE);
        }
        return roleDO;
    }

    /**
     * 获得自身的代理对象，解决 AOP 不生效问题
     *
     * @return 自身的代理对象
     */
    private RoleServiceImpl getSelf() {
        return SpringUtils.getBean(getClass());
    }

}
